split data into train validation and test